home *** CD-ROM | disk | FTP | other *** search
/ PC World 2007 March / PCWorld_2007-03_cd.bin / v cisle / httrack / httrack-3.41-2.exe / {app} / src_win / WinHTTrackIEBar / WinHTTrackLauncher.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2004-02-15  |  6.0 KB  |  229 lines
  1. // WinHTTrackLauncher.cpp : Implementation of CWinHTTrackIEBarApp and DLL registration.
  2.  
  3. #include "stdafx.h"
  4. #include "WinHTTrackIEBar.h"
  5. #include "WinHTTrackLauncher.h"
  6.  
  7. extern "C" {
  8. #include <direct.h>
  9. };
  10.  
  11. #include <commctrl.h>
  12. #include <mshtml.h>
  13.  
  14. #include "projectInfo.h"
  15.  
  16. /////////////////////////////////////////////////////////////////////////////
  17. //
  18.  
  19. static char* BSTRtoPCHAR(BSTR str, UINT codepage = CP_ACP) {
  20.   if (str != NULL) {
  21.       int multiByteLength = ::WideCharToMultiByte( codepage, 0, str, wcslen(str), NULL, NULL, NULL, NULL); 
  22.     if (multiByteLength > 0) 
  23.     { 
  24.       char *pcharMutiByteBuffer = (char*) malloc(multiByteLength + 1); 
  25.       ::WideCharToMultiByte( codepage, 0, str, wcslen(str), pcharMutiByteBuffer, multiByteLength, NULL, NULL); 
  26.       pcharMutiByteBuffer[multiByteLength] = '\0';
  27.       return pcharMutiByteBuffer;
  28.     }
  29.   }
  30.   return NULL;
  31. }
  32.  
  33. static int fexist(char* s) {
  34.   FILE* fp = fopen(s, "rb");
  35.   if (fp) fclose(fp);
  36.   return fp != NULL;
  37. } 
  38.  
  39. STDMETHODIMP WinHTTrackLauncher::InterfaceSupportsErrorInfo(REFIID riid)
  40. {
  41.   static const IID* arr[] = 
  42.   {
  43.     &IID_IWinHTTrackLauncher,
  44.   };
  45.   
  46.   for (int i=0;i<sizeof(arr)/sizeof(arr[0]);i++)
  47.   {
  48.     if (InlineIsEqualGUID(*arr[i],riid))
  49.       return S_OK;
  50.   }
  51.   return S_FALSE;
  52. }
  53.  
  54. // IOleCommandTarget
  55. STDMETHODIMP WinHTTrackLauncher::QueryStatus(const GUID __RPC_FAR *pguidCmdGroup, ULONG cCmds, OLECMD __RPC_FAR prgCmds[  ], OLECMDTEXT __RPC_FAR *pCmdText) {
  56.     HRESULT hr = E_POINTER;
  57.  
  58.     if ( prgCmds != NULL)
  59.     {
  60.         for ( ULONG i = 0; i < cCmds; i++)
  61.             prgCmds[i].cmdf = OLECMDF_SUPPORTED | OLECMDF_ENABLED; 
  62.         
  63.         hr = S_OK;
  64.     }
  65.     return hr;
  66. }
  67.  
  68. STDMETHODIMP WinHTTrackLauncher::Exec(const GUID *pguidCmdGroup, DWORD nCmdID, DWORD nCmdExecOpt, VARIANTARG *pvaIn, VARIANTARG *pvaOut) {
  69.   HKEY key;
  70.   DWORD type;
  71.   char path[256];
  72.   char appPath[256];
  73.   DWORD size = sizeof(path) - 2;
  74.   BSTR url = NULL;
  75.   BSTR title = NULL;
  76.   char* urlPCHAR = NULL;
  77.   char* titlePCHAR = NULL;
  78.  
  79.   if (RegOpenKeyExA(HKEY_CURRENT_USER, "Software\\WinHTTrack Website Copier\\WinHTTrack Website Copier\\DefaultValues",
  80.     0, KEY_READ,
  81.     &key) == ERROR_SUCCESS) {
  82.     if (RegQueryValueExA(key, "BasePath", 0, &type,
  83.       (LPBYTE) path, &size) == ERROR_SUCCESS)
  84.     {
  85.       path[size] = '\0';
  86.     }
  87.     RegCloseKey(key);
  88.   }
  89.   size = sizeof(appPath) - 2;
  90.   if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, "Software\\WinHTTrack Website Copier\\WinHTTrack Website Copier",
  91.     0, KEY_READ,
  92.     &key) == ERROR_SUCCESS) {
  93.     if (RegQueryValueExA(key, "Path", 0, &type,
  94.       (LPBYTE) appPath, &size) == ERROR_SUCCESS)
  95.     {
  96.       appPath[size] = '\0';
  97.       strcat(appPath, "\\WinHTTrack.exe");
  98.     }
  99.     RegCloseKey(key);
  100.   }
  101.   
  102.   if (path[0] != '\0' && appPath[0] != '\0') {
  103.  
  104.     CComPtr<IDispatch> doc;
  105.     if (SUCCEEDED(browser->get_Document(&doc))) {
  106.       CComQIPtr<IHTMLDocument2> hDoc(doc);
  107.       if (hDoc != NULL) {
  108.         if (SUCCEEDED(hDoc->get_title(&title))) {
  109.           titlePCHAR = BSTRtoPCHAR(title);
  110.         }
  111.       }
  112.     }
  113.     if (SUCCEEDED(browser->get_LocationURL(&url))) {
  114.       urlPCHAR = BSTRtoPCHAR(url);
  115.     }
  116.     
  117.     if (urlPCHAR != NULL && titlePCHAR != NULL) {
  118.       int i, j;     
  119.       char lastCh = 0;
  120.       for(i = 0 ; i < (int) strlen(titlePCHAR) ; i++) {
  121.         if (titlePCHAR[i] == '\"') {
  122.           titlePCHAR[i] = '\'';
  123.         }
  124.         else if (titlePCHAR[i] == ',' 
  125.           || titlePCHAR[i] == '~'
  126.           || titlePCHAR[i] == '\\'
  127.           || titlePCHAR[i] == '/'
  128.           || titlePCHAR[i] == ':'
  129.           || titlePCHAR[i] == '*'
  130.           || titlePCHAR[i] == '?'
  131.           || titlePCHAR[i] == '<'
  132.           || titlePCHAR[i] == '>'
  133.           || titlePCHAR[i] == '|'
  134.           || ((unsigned char)titlePCHAR[i]) < 32
  135.           ) {
  136.               titlePCHAR[i] = ' ';
  137.           }
  138.       }
  139.       for(i = 0, j = 0 ; titlePCHAR[i] != '\0' ; i++) {
  140.           if (titlePCHAR[i] != ' ' || lastCh != ' ') {
  141.               if (i != j) {
  142.                   titlePCHAR[j++] = titlePCHAR[i];
  143.               }
  144.               lastCh = titlePCHAR[i];
  145.           }
  146.       }
  147.       titlePCHAR[j++] = '\0';
  148.       if (strlen(titlePCHAR) > 128)
  149.           titlePCHAR[128] = '\0';
  150.  
  151.       _bstr_t titleWtl = titlePCHAR;
  152.       CprojectInfo info(&titleWtl);
  153.       if (info.DoModal() == IDOK) {
  154.           /* Form name */
  155.           strcat(path, "\\");
  156.           strcat(path, (char*)titleWtl);
  157.           int backPos = strlen(path);
  158.           strcat(path, ".whtt");
  159.           /* Open */
  160.           if (!fexist(path)) {
  161.               FILE* fp = fopen(path, "wb");
  162.               if (fp != NULL) {
  163.                   fclose(fp);
  164.               }
  165.               path[backPos] = '\0';
  166.               _mkdir(path);
  167.               strcat(path, "\\hts-cache");
  168.               _mkdir(path);
  169.               strcat(path, "\\winprofile.ini");
  170.               fp = fopen(path, "wb");
  171.               if (fp != NULL) {
  172.                   fprintf(fp, "CurrentUrl=%s\r\n", urlPCHAR);
  173.                   fprintf(fp, "CurrentAction=1\r\n");
  174.                   fclose(fp);
  175.               }
  176.               path[backPos] = '\0';
  177.               strcat(path, ".whtt");
  178.           }
  179.           if (fexist(path)) {       
  180.               if (!ShellExecuteA(NULL, "open", path, "", "", SW_RESTORE)) {
  181.               }
  182.           }
  183.       }
  184.  
  185.     } else
  186.       MessageBox(NULL, L"Can not find any URL", L"WinHTTrack", MB_SETFOREGROUND);
  187.   }
  188.   if (url != NULL) {
  189.     SysFreeString(url);
  190.   }   
  191.   if (title != NULL) {
  192.     SysFreeString(title);
  193.   }
  194.   if (urlPCHAR != NULL) {
  195.     free(urlPCHAR);
  196.   }
  197.   if (titlePCHAR != NULL) {
  198.     free(titlePCHAR);
  199.   }
  200.   return S_OK;
  201. }
  202.  
  203. // IObjectWithSite - See KB257717
  204. STDMETHODIMP WinHTTrackLauncher::SetSite(IUnknown* pClientSite) {
  205.   HRESULT hr = S_OK;
  206.   CComPtr<IServiceProvider> isp, isp2;
  207.   if (pClientSite != NULL)
  208.   {
  209.     if (SUCCEEDED(pClientSite->QueryInterface(IID_IServiceProvider, reinterpret_cast<void **>(&isp))))
  210.     {
  211.       if (SUCCEEDED(isp->QueryService(SID_STopLevelBrowser, IID_IServiceProvider, reinterpret_cast<void **>(&isp2))))
  212.       {
  213.         if (SUCCEEDED(isp2->QueryService(SID_SWebBrowserApp, IID_IWebBrowser2, reinterpret_cast<void **>(&browser))))
  214.         {
  215.         }
  216.       }
  217.     }
  218.   }
  219.   return S_OK;
  220. }
  221.  
  222. STDMETHODIMP WinHTTrackLauncher::GetSite(REFIID riid, void** ppvSite) {
  223.     if (NULL != ppvSite) {
  224.         *ppvSite = browser;
  225.     }
  226.     return E_NOTIMPL;
  227. }
  228.  
  229.